//
//  UIButton+countDown.m
//  NetworkEgOc
//
//  Created by iosdev on 15/3/17.
//  Copyright (c) 2015年 iosdev. All rights reserved.
//

#import "UIButton+CountDown.h"
#import <objc/runtime.h>
static const void *IndieBandNameKey = &IndieBandNameKey;

@implementation UIButton (countDown)


- (dispatch_source_t)timer {
    return objc_getAssociatedObject(self, IndieBandNameKey);
}

- (void)setTimer:(dispatch_source_t)timer{
    objc_setAssociatedObject(self, IndieBandNameKey, timer, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}


-(void)startTime:(NSInteger )timeout title:(NSString *)tittle
{
    __block NSInteger timeOut=timeout; //倒计时时间
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    self.timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0,queue);
    dispatch_source_set_timer(self.timer,dispatch_walltime(NULL, 0),1.0*NSEC_PER_SEC, 0); //每秒执行
    dispatch_source_set_event_handler(self.timer, ^{
        if(timeOut<=0){ //倒计时结束，关闭
            dispatch_source_cancel(self.timer);
            dispatch_async(dispatch_get_main_queue(), ^{
                //设置界面的按钮显示 根据自己需求设置
                [self setTitle:tittle forState:UIControlStateNormal];
                self.enabled = YES;

            });
        }else{
            //            int minutes = timeout / 60;
            int seconds = timeOut % 60;
            NSString *strTime = [NSString stringWithFormat:@"%.2d", seconds];
            dispatch_async(dispatch_get_main_queue(), ^{
                //设置界面的按钮显示 根据自己需求设置
                [self setTitle:[NSString stringWithFormat:@"%@(%@s)",tittle,strTime] forState:UIControlStateDisabled];
                self.enabled = NO;
                
            });
            timeOut--;
        }
    });
    dispatch_resume(self.timer);
    
}

- (void)dealloc
{
    if (self.timer) {
        dispatch_source_cancel(self.timer);
    }
}

@end
